import numpy as np
import faiss
import time
import faiss_learn


all_feature= np.load('all_id_8000.npy').astype('float32')